tensor([[[0.3900, 0.3900],
         [0.3900, 0.3900]],

        [[0.2800, 0.2800],
         [0.2800, 0.2800]],

        [[0.3300, 0.3300],
         [0.3300, 0.3300]]], grad_fn=<RoundBackward1>)